Done with the classes, just need to add comments.