need to add hella __main__ tests