some styling stuff and added caption functionality