Made changes to make the code more clear