Adding changes back in