Cleaned up the main loop and made prompts accurate
Placed the reset instruction in the right place
Replaced direct square accesses with function call