bug fix, small feature added