Refactor the code basd to include further functionality