fix functionality
Merge pull request #4 from FWOliver/fix_functionality

fix functionality