The logic for 2-bit is in place but incorrect.