Change findById function