Comment fix - add @return tag