Added the data back in

end screen is fully functional :)