refactoring
adding basic ui