Add formatting in test command