Add lab01 code