Edited the run method