#8 Fixed a little the UI example