Refactor U class exercise - Modularity 1 Class Exercise