Update lab8-4 with correct code