Just made a small change
Small change to the code