Fixed code to use a different cost function. Discn with Shalmoli