Lab 4 seems to be working