Got cellSize to work as a separate method, refactored code according to code review from Jon to make it cleaner