Add lab 3 solution
Modify return values