updated to add info screen