Actually remembers selection now. Oops