added more code for statement handling