added small functionality