Add member function to set width