Add the missing lab6 code