support custom margin and adding a button

Signed-off-by: Marc Lefrançois <mlefrancois@mirego.com>