added end game logic, also fixed hitting and staying