Finishing up Java class