Improved background size, start adding types to docstring