adding code alex wants me to get working