Add comments and demo code in readme