Added import to fix bug
Added functionality for users to choose what they are tested on