Title: Handle the Loss