WIP: Updating shader to be fully programmable