beatifiying code + giving option to choose tests slectively