Fixed small bugs. IT WORKSgit