Added classes to src file

Done to tidy work space.