Add label, find with label and switch Main code to JUnit tests