Edited the  code ;P