Added tests for models & deleting code line(as comments)