Started on model creation but ran into logic flaw