Adding in shader state saving functionality in the only way I know how