Setup UI for code running