Updated with @ogrisel's helpful comments.