Re-added the gen test
Added better support for compute shaders
Added null checks utility methods