cleaned up imports
removed block keyword from plotting but use interactive mode instead