fixed the cost function in warmup, currently working on part_three of homework6. Generic checkpoint