Still debugging. Good solution for small dataset: 0203.