Added return key functionality