Changed result screen to have two buttons