ass4 code done