bug fix + enhancement

fixed bug (i#3). enhancement (i#1)