Change classes to case classes!