Changed a misleading comment.
Added the balance check for S also to python