Cleaned

Added a few more comments. Removed unnecessary imports and variables.