Minor change
Slight code modification