Add menu / Fix bugs