Added other models, and this commit makes it work with runserver