Finally fixed the learning