Cleaned up code. Removed unused classes. Updated comments.
Updated more documentation