Updates to Unit class