Codes for Lab 2 updated