Code from the second demo. Done!