Changes per feedback on #23