If $A$ is a non-null diagonal matrix of order $3$ such that $A^4 = A^2$, then the number of possible matrices $A$ are