fix functionality

well functioning