fixes issue #1

Added code.