re-did __main__ for testing locally