Add back uniform location caching (speed up render)

Add back shader uniform location caching with a cleaner class (can speed up render on certain devices)