Tweaked test.jl file