Put model into own class for easier use in service