Set up proper header and cpp files for classes. Thanks Chris Grandin