added unit 4 ( and unit 2 ) tests