Resolved Issue 1

Now correctly gets imported functions, classes.
Now properly catches printed output