Added class in order to make this a real project.