added abillity to specify a unit