Modified junit file