Lab 1 code