Add highlighting and amsthm package