Removed some inrelevant stuff from the docs
#1 Moved models to different package. Updated tests