improve code. Fix typos