Fix #5 ; Fix #29 ; Bugs & UI fix