Add some sample code for expansion toy