also should include code for final game :)