Minor fix in main script
Game functioning, some minor fixes outstanding