Added framework for homework-4