Added back setting caption which by mistake was removed #1