Resetting the game now works properly
Added comments and tidied some things up.