(IMPROVE) Implement ability to  be able to take in port as a parameter