Still need to add the ability to generate test cases