Adding overlay. TODO logic and buttons