Added a small feature