Small api change
I'm smart
Merge pull request #15 from Cal-CS-61A-Staff/models

Small api change