Separate testing case from main