Add bg image and clean up code