Initial code for labwork 3