Potential final code for first demo