Improve code comments and update README