Added short code function

Still needs some customization