Lab 4 tested and working fully