Fixes a z-index bug
Updated TODO