Working code for "First test passing"