Close #3

#3 fixed