Removed print statements, fixed loading of the top-level module