Tryin to fix up this modal stuff