Updated code lab 2.