Separated model into it's own file