Added functions for setting unit and model