Add extra functionality