Revert "removed unwanted variable from models"

This reverts commit d44d736b18191cbf6791a05cccf9fa755f775e7f.