Update code style. Add comments.