added unit 2 assignment