Simplify UBO init/update