minor edit to unit3