Adding my homework finally