Adding set up code