Refactor and added UI based on scene2d