#8 added classes to api models file