Cleaned up the main computer program