adapt with suggestions from @okomarov

* simplify code
* remove "inner sep"