Add annotations @SuppressWarnings("unchecked") for 2 classes