Code should function as it did before