Fix for #215 and some other related improvements

looks like so