Adding code to src files and design