support if-expressions
add missing import
support class and module docstrings
support set literals