Changes per feedback on #242