Removed unused imports, cleaned up some code, updated README file.