Changed Shader class to nested class within entity