Add more postprocessing
Add space dependent mass