Add code solution for first demo