Add test function to ghci