Updated minor bug and tested.