Add better shader and update transformation matrix code.