Trying to solve the same lab on level B.