Works with views, need to make proper Viewmodel