some changes to submit my code. need to add images later