Adding edit view and styling everything