updated final lab code