Fix exit function, visual fix