Definitely need to improve this further. I keep screwing up at the end a lot.