Saddle points exercise | 1st iteration -test_code