Lab 04 code