Cleaning up main function

Agent walks forward but that was just a simple test