Cleaned the milestone controller