Basic logic working. Needs to be styled.