Updating the source file with my solution