Cleaned up pulse latch and mux