Fixed wrong configuration in draw_states.m