Minor changes. Fixed a small bug.