Working on the SQL Alchemy models