Fix bug and add .gitgnore