Going to post an answer as the comment code is a bit messy