Add logic to set capacity for the station