Refactored code.
Game finishes feature defined. Still in red