Added 3rd test + fixed sum function