Simple program to study reinforcement learning