Assignment 0: Fix javadoc @return tags and remove unused import