pulled notebook and converted it to a .py file