extracted a couple methods and fixed a bug where my if display statement wasn't functioning