Added more documentation to the code...

...changed my classes to modules.