added model stuff