Updates to lab5.c

More code!! Whoooo!