Refactor test CLI and allow for more languages than python