Updated run method