Almost done with core logic. Need to implement restart of game