Updated demo in accordance to the library