Adding changes to base model