Fixed bug in byes classifier