Adding support to run test within IDEA