My original code