Add classes

Add classes to reflect the db schema