Milestone 2 (optimize NDCG@5) final