added more logoc for game