Updated lab4 script