For demo purposes hacked together some stuff