Cleaned up shader code