Add simple validation logic. Will add more later.