Removed my comments and cleaned up for PEP8