Add comments within code

Fix #8