Add decoration. Closes #11