Moved model init/drawing to own class