functionality working, now need to do styling