updated 2 sections. Still need design.. not compiling at the moment. Don't know why