Update models.py

corrected sets to be lists because i'm awful at python