Tiny update, hopefully caption looks nicer