fix something and add milestone code