refactor: replace WIDTH and HEIGHT const with SQUARE