Start work on ui part