modified code to include key part of testing... moron