change code to a setup to test