small functionality was added