Changes per feedback on #4