Edited the code