Refactored the code added capacity to airport