Added comments & cleaned up code
Version: 0.0.1