Remove unneeded print statement, change __init__ to simplify import statements