forked and customized it a bit