Improved the code readability as Michael suggested.