Made some changes for next lab