Some shader changes for Uniform Buffer