fixing typos
adding more tests