Lab 4 initial working solution