Separating Main Code from Unit Test Code