Minor fix
Adding features to TextUI