change cost function
- normalize x_final before computing cost function
- use negative of dot product instead of L2
bigger figure size