Move color logic to interactive shell