adding not to test cases and fixing wagon class