Small changes to lab5