Cleaned up shader