Looks ok.. Take off the gradients?