Changed some stuff in the current models and added comment.py