Minor change

Started thinking of making widget for this.