very hacky but it'll work for the demo