Fixed bug and added demo gif