Move classes here from example project