Added done message at the end again