Added code for lab2