working on bug3

#9
fixing bug

fix #9
change