Adding command functionality