tests refactored
Added capacity to the station