Update Trial Problem C

Updated the code - it actually works now :p