FIX: No need for __main__