fixed small bug and ui improvement
Release v0.1.4