made more changes after finishing the lab