Fix the classes.
Adding proper documentation.