oh dear, I should be committing way more often
finished code for code test