Better logging, numerical gradient checking, visualization for all layers and little error fix