Added header file + more code

Added header file with classes. A couple pictures added to for future
use.