Added comment for clarity where model can be improved