Imported draw code for Stage to new framework.