Adding reflection for unit 4