README image and better comments