minor changes to Lab5