Added model stuff