Add code from 3 lab