fix #6 support returning statements