Update models
Quick fix for pep8