Made the add units a little more user friendly