added choosing initial statement option, some changes needed