Fixed issue #1

Going to fix issue #1