Add cancel button & button styling/fixes