add note
add some comments for future improvements
fix equations
fix typo