I don't really know what I'm doing!

Added some classes from the example code.