Switched from assembly plugin to shade plugin for better flexibility