Working on UI  and gameplay again