Added \usepackage{caption}