modified file for issue #5