Minor changes during demo to team