Implemented suggested change