Added media and a slight bug fix