Edited original file with solution