Allow for multiple entities
Push shader up to entity level. Will come up with a better way of managing it later.