Tidy up shader