added test + made it pass