Main UI finally figured out