Added better UI components, levels