Modularized code by moving display to its own source file