Cleaned up notebook