Kangana wraps up Thalaivi schedule