trying to fix issue #5 part 2