Tidying up Java side of Game Screen