TODO: add second screen