cleaned up main, added return values within loop for better control and continuance
removed major whitespaces in main ...