Working code according to tests