Added button and fixed up playScreen