Python return statement not returning correct output