Add shader handling class