Made changes that don't completely work properly for demo purposes. Will fix later.