Got compute shader to work :)