Adding back some of your original code that was accidentally deleted...