Addressed issue #3

Signed-off-by: David Dorfman <wiz3kid@gmail.com>