edited the unit 2 further